package com.softeem.utils;

import java.util.Random;


public class RandomCodeUtils {

    /**
     * 随机生成验证码
     *
     * @param length 长度为4位或者6位
     * @return
     */
    public static Integer getRandomCodes(int length) {

        Integer code = null;

        if(length == 4) {
            code = new Random().nextInt(9999);//生成随机数，最大为9999

            if(code < 1000) {

                code = code + 1000;//保证随机数为4位数字
            }

        } else if(length == 6) {

            code = new Random().nextInt(999999);//生成随机数，最大为999999

            if(code < 100000) {
                code = code + 100000;//保证随机数为6位数字
            }

        } else {

            throw new RuntimeException("只能生成4位或6位数字验证码");

        }
        return code;
    }



    public static void main(String[] args) throws Exception {

        Integer randomCodes = RandomCodeUtils.getRandomCodes(4);
        System.out.println(randomCodes);

    }
}